import os
import hashlib
import zipfile
import yaml

def md5_code(passwd):
    md5 = hashlib.md5()
    md5. update (passwd. encode("utf-8"))
    return md5.hexdigest ()

def trans_str(text):
    return text.replace('":"','=').replace('":','=')

def read_yaml(yaml_file):
    with open(yaml_file,encoding="utf-8") as f:
         return yaml.safe_load(f)

def zip_dir(startdir,zip_file):
    # 压缩文件夹
    _zip_file = os.path. join(startdir,zip_file)
    if os.path.exists(_zip_file):
        #先删除
        os.remove(_zip_file)
    z = zipfile.ZipFile(_zip_file,'w',zipfile.ZIP_DEFLATED)
    for dir_path, dir_names,file_names in os. walk(startdir):
        fpath = dir_path.replace(startdir,'')
        # 不replace的话，就从根目录开始复制
        fpath = fpath and fpath + os. sep or''
        for filename in file_names:
            z. write(os. path. join(dir_path, filename), fpath + filename)
    z.close ()